# CUDA 版本对应的 PyTorch、TorchVision、TorchAudio 版本映射
# 以及对应的 CUDA 支持标签
TORCH_VERSION_MAP = {
    '10.2': ('1.13.1', '0.14.1', '0.13.1', 'cu102'),
    '11.3': ('1.13.1', '0.14.1', '0.13.1', 'cu113'),
    '11.6': ('1.13.1', '0.14.1', '0.13.1', 'cu116'),
    '11.7': ('2.0.1', '0.15.2', '0.15.1', 'cu117'),
    '11.8': ('2.1.2', '0.16.2', '2.1.2', 'cu118'),
    '12.1': ('2.2.2', '0.17.2', '2.2.2', 'cu121'),
    '12.4': ('2.3.1', '0.18.1', '2.3.1', 'cu121'),  # cu121 兼容 12.4
    '12.5': ('2.4.1', '0.19.1', '2.4.1', 'cu121'),
}